package tech.aiflowy.common.ai;

import com.agentsflex.core.document.DocumentParser;
import com.agentsflex.core.document.parser.DefaultTextDocumentParser;
import com.agentsflex.document.parser.PdfBoxDocumentParser;
import com.agentsflex.document.parser.PoiDocumentParser;

public class DocumentParserFactory {

    public static DocumentParser getDocumentParser(String typeOrFileName) {
        if (typeOrFileName == null) {
            throw new NullPointerException("typeOrFileName can not be null");
        } else {
            typeOrFileName = typeOrFileName.trim().toLowerCase();
        }
        if (typeOrFileName.endsWith(".pdf")) {
            return new PdfBoxDocumentParser();
        }
        if (typeOrFileName.endsWith(".docx")) {
            return new PoiDocumentParser();
        }
        if (typeOrFileName.endsWith(".txt")) {
            return new DefaultTextDocumentParser();
        }
        if (typeOrFileName.endsWith(".md")) {
            return new MarkdownDocumentParser();
        }
        if (typeOrFileName.endsWith(".xlsx")) {
            return new ExcelDocumentParser();
        }
        return null;
    }
}
